iT邦幫忙

2022 iThome 鐵人賽

DAY 11
0

根據訓練集放到建立好的模型裡,看其預測的機率值,
由於先前有將其做標準化,因此,並將須將結果轉換回股價。
同理,測試資料集一樣會換回股價,

train_predict=model.predict(X_train)
test_predict=model.predict(X_test)
train_predict=scaler.inverse_transform(train_predict)
test_predict=scaler.inverse_transform(test_predict)

最後檢視模型的均方誤,只不過測試集的結果,相比訓練集通常會更偏差一些。

import math
from sklearn.metrics import mean_squared_error
math.sqrt(mean_squared_error(y_train,train_predict))

參考:https://ithelp.ithome.com.tw/articles/10279778


上一篇
股價預測篇-建模part2(金融篇)
下一篇
股價預測篇-爬蟲part5(金融類)
系列文
從無到有,爬蟲-分析-預測建模,把實務面常見問題逐一釐清18
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言